package com.shudingke.utils;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.github.pagehelper.PageInfo;
import org.apache.poi.ss.formula.functions.T;

import java.util.ArrayList;
import java.util.List;

public class MyPageList {
    public static <T>List<Integer> getPageList(String page, PageInfo<T> info){
        List<Integer> list = new ArrayList<>();
        if (info.getList() == null){
            return null;
        }
        if (info.getPages() <= 5){
            for (int i = 1; i <= info.getPages(); i++ ){
                list.add(i);
            }
        }else {
            if (Integer.parseInt(page) + 4 < info.getPages()){
                for (int i = Integer.parseInt(page); i <= Integer.parseInt(page) + 4; i++ ){
                    list.add(i);
                }
            }else {
                for (int i = info.getPages() - 4; i <= info.getPages(); i++ ){
                    list.add(i);
                }
            }
        }
        
        return list;
    }
}
